#该问题发生于我需要将yolo中的某个标签替换为另一个标签
import argparse
from pathlib import Path
import cv2
import xml.etree.ElementTree as ET


def Parse_Arguments():
    parser = argparse.ArgumentParser(description="根据键盘内容改变xml内容")
    parser.add_argument('--source_xml_dir', type=str, default=r"C:\Users\luoluoluo\Desktop\xml07")
    return parser.parse_args()

if __name__ == "__main__":
    args = Parse_Arguments()
    source_xml_dir = Path(args.source_xml_dir).glob("*.xml")
    for xml_path in source_xml_dir:
        tree = ET.parse(xml_path)
        root = tree.getroot()
        for obj in root.findall('object'):
            # 检查<name>标签的内容
            if obj.find('name').text == 'head':
                # 修改<name>标签的内容
                obj.find('name').text = 'vest'  # 你可以在这里指定新的值
        tree.write(xml_path, encoding='utf-8', xml_declaration=True)
